package com.practice;

/**
 * @author YJ
 * @data 2025/6/27 17:26
 */
public class IsBalancedBinaryTree {
    /**
     * 输入一棵节点数为 n 二叉树，判断该二叉树是否是平衡二叉树。
     * 在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树
     * 平衡二叉树（Balanced Binary Tree），具有以下性质：
     * 它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。
     */
    public static boolean isBalancedBinaryTree(TreeNode root) {
        //空树也是平衡二叉树
        if (root == null) return true;
        return getDepth(root) != -1;
    }

    public static int getDepth(TreeNode root) {
        if (root == null) return 0;
        //计算当前root左子树深度差
        int left = getDepth(root.left);
        //当前root左子树不平衡，则该树不平衡
        if (left < 0) {
            return -1;
        }
        //计算当前root右子树深度差
        int right = getDepth(root.right);
        //当前root右子树不平衡，则该树不平衡
        if (right < 0) {
            return -1;
        }
        //计算深度差
        return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
    }
}
